package com.JingYan.util.json;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.alibaba.fastjson2.JSONValidator;


import java.util.Iterator;
import java.util.Map;

/**
 * @author DongJiaQi
 * @create 2022/8/5 10:16
 * @Description: fastjson 去除前后空格
 */
public class JsonTrimUtils {
    public JsonTrimUtils() {
    }
    /**
     * 去除json值前后空格
     * @param jsonStr jsonStr
     * @return
     */
    public static JSON jsonTrim(String jsonStr) {
        JSONValidator.Type type = getJsonType(jsonStr);
        if (type == JSONValidator.Type.Object || type == JSONValidator.Type.Value) {
            return jsonTrim(JSONObject.parseObject(jsonStr));
        } else if (type == JSONValidator.Type.Array) {
            JSONArray array = JSONArray.parseArray(jsonStr);
            jsonTrim(array);
            return array;
        }
        return null;
    }

    /**
     * 获得Json的类型
     * @param jsonStr
     * @return
     */
    public static JSONValidator.Type getJsonType(String jsonStr) {
        JSONValidator.Type type = JSONValidator.from(jsonStr).getType();
        return type;
    }

    /**
     * 去除json值前后空格
     * @param json jsonStr
     * @return
     */
    public static JSON jsonTrim(JSON json) {
        if (json instanceof JSONObject) {
            return  jsonTrim((JSONObject)json);
        } else if (json instanceof JSONArray) {
            jsonTrim((JSONArray) json);
            return json;
        }
        return null;
    }
    /**
     * 去除value的空格
     *
     * @param jsonObject jsonObject
     * @return
     */
    public static JSONObject jsonTrim(JSONObject jsonObject) {
        Iterator<Map.Entry<String, Object>> iterator = jsonObject.entrySet().iterator();
        while (iterator.hasNext()) {
            Map.Entry<String, Object> next = iterator.next();
            Object value = next.getValue();
            if (value != null) {
                if (value instanceof String) {
                    //清空值前后空格
                    jsonObject.put(next.getKey(), ((String) value).trim());
                } else if (value instanceof JSONObject) {
                    jsonTrim((JSONObject) value);
                } else if (value instanceof JSONArray) {
                    jsonTrim((JSONArray) value);
                }
            }
        }

        return jsonObject;
    }
    /**
     * 清空JSONArray 值前后空格
     * @param array
     */
    private static void jsonTrim(JSONArray array) {
        if (array.size() > 0) {
            for (int i = 0; i < array.size(); i++) {
                Object object = array.get(i);
                if (object != null) {
                    if (object instanceof String) {
                        array.set(i, ((String) object).trim());
                    } else if (object instanceof JSONObject) {
                        jsonTrim((JSONObject) object);
                    } else if (object instanceof JSONArray) {
                        jsonTrim((JSONArray) object);
                    }
                }
            }
        }
    }
}
